from BaseNN import nn
model = nn()
checkpoint = 'checkpoints/basenn/1.pth'
out_file='checkpoints/basenn/basenn.onnx'
model.convert(checkpoint=checkpoint, out_file=out_file)